from sqlalchemy.orm import Session
from ecg.db import SessionLocal  # 假设你在 ecg/db.py 中配置了数据库连接
from ecg.models import ECGRecord  # 假设你在 ecg/models.py 中定义了 ECGRecord 模型
from devices.models import Device  # 假设你有一个设备模型
from datetime import datetime

# 获取数据库会话
db = SessionLocal()

# 查询设备（假设设备名称是 'nb'，设备编号是 'ECG1'）
device = db.query(Device).filter(Device.name == 'nb', Device.device_id == 'ECG1').first()

if device:
    # 创建心电数据记录
    ecg_data = ECGRecord(
        device_id=device.id,  # 插入设备的 ID
        heart_rate=72.5,  # 设定心率，假设为 72.5
        data="{'ecg': 'data goes here'}",  # 假设原始心电数据
        timestamp=datetime.utcnow()  # 使用当前时间戳
    )

    # 将数据添加到会话中并提交
    db.add(ecg_data)
    db.commit()

    print("心电数据已成功插入到数据库！")
else:
    print("设备未找到，请检查设备编号或名称。")

# 关闭数据库会话
db.close()
